Average marks program in Python